#include "../mqserver/mq_route.hpp"
#include <gtest/gtest.h>

class RouteTest : public testing::Environment
{
    virtual void SetUp()
    {
    }

    virtual void TearDown()
    {
        // mqmp->clear();
    }
};

TEST(route_test, legal_routing_key)
{
    std::string rkey1 = "news.music.pop";
    std::string rkey2 = "news..music.pop";
    std::string rkey3 = "news.,music.pop"; // fasle
    std::string rkey4 = "news.music_123.pop";
    ASSERT_EQ(bitmq::Route::isLegalRoutingKey(rkey1), true);
    ASSERT_EQ(bitmq::Route::isLegalRoutingKey(rkey2), true);
    ASSERT_EQ(bitmq::Route::isLegalRoutingKey(rkey3), false);
    ASSERT_EQ(bitmq::Route::isLegalRoutingKey(rkey4), true);
}

TEST(route_test, legal_binding_key)
{
    std::string bkey1 = "news.music.pop";
    std::string bkey2 = "news.#.music.pop";
    std::string bkey3 = "news.#.*.music.pop";     // false
    std::string bkey4 = "news.#.#.music_123.pop"; // false
    std::string bkey5 = "news.*.*.music_123.pop";
    std::string bkey6 = "news.,music_123.pop"; // false
    ASSERT_EQ(bitmq::Route::isLegalBindingKey(bkey1), true);
    ASSERT_EQ(bitmq::Route::isLegalBindingKey(bkey2), true);
    ASSERT_EQ(bitmq::Route::isLegalBindingKey(bkey3), false);
    ASSERT_EQ(bitmq::Route::isLegalBindingKey(bkey4), false);
    ASSERT_EQ(bitmq::Route::isLegalBindingKey(bkey5), true);
    ASSERT_EQ(bitmq::Route::isLegalBindingKey(bkey6), false);
}

// [测试⽤例]
// binding key routing key result
// aaa aaa true
// aaa.bbb aaa.bbb true
// aaa.bbb aaa.bbb.ccc false
// aaa.bbb aaa.ccc false
// aaa.#.bbb aaa.bbb.ccc false
// aaa.bbb.# aaa.ccc.bbb false
// #.bbb.ccc aaa.bbb.ccc.ddd false
// aaa.bbb.ccc aaa.bbb.ccc true
// aaa.* aaa.bbb true
// aaa.*.bbb aaa.bbb.ccc false
// *.aaa.bbb aaa.bbb false
// # aaa.bbb.ccc true
// aaa.# aaa.bbb true
// aaa.# aaa.bbb.ccc true
// aaa.#.ccc aaa.ccc true
// aaa.#.ccc aaa.bbb.ccc true
// aaa.#.ccc aaa.aaa.bbb.ccc true
// #.ccc ccc true
// #.ccc aaa.bbb.ccc true
// aaa.#.ccc.ccc aaa.bbb.ccc.ccc.ccc true
// aaa.#.bbb.*.bbb aaa.ddd.ccc.bbb.eee.bbb true

TEST(route_test, legal_route_key)
{
    std::vector<std::string> bkeys = {
        "aaa",
        "aaa.bbb",
        "aaa.bbb",
        "aaa.bbb",
        "aaa.bbb.ccc",
        "aaa.*",
        "aaa.*.bbb",
        "*.aaa.bbb",
        "#",
        "aaa.#",
        "aaa.#",
        "aaa.#.ccc",
        "aaa.#.ccc",
        "aaa.#.ccc",
        "#.ccc",
        "#.ccc",
        "aaa.#.ccc.ccc",
        "aaa.#.bbb.*.bbb"};
    std::vector<std::string> rkeys = {
        "aaa",
        "aaa.bbb",
        "aaa.bbb.ccc",
        "aaa.ccc",
        "aaa.bbb.ccc",
        "aaa.bbb",
        "aaa.bbb.ccc",
        "aaa.bbb",
        "aaa.bbb.ccc",
        "aaa.bbb",
        "aaa.bbb.ccc",
        "aaa.ccc",
        "aaa.bbb.ccc",
        "aaa.aaa.bbb.ccc",
        "ccc",
        "aaa.bbb.ccc",
        "aaa.bbb.ccc.ccc.ccc",
        "aaa.ddd.ccc.bbb.eee.bbb"};

    std::vector<bool> result = {
        true,
        true,
        false,
        false,
        true,
        true,
        false,
        false,
        true,
        true,
        true,
        true,
        true,
        true,
        true,
        true,
        true,
        true};

    for (int i = 0; i < bkeys.size(); i++)
    {
        ASSERT_EQ(bitmq::Route::route(bitmq::ExchangeType::TOPIC, rkeys[i], bkeys[i]), result[i]);
    }
}
int main(int argc, char *argv[])
{
    testing::InitGoogleTest(&argc, argv);
    testing::AddGlobalTestEnvironment(new RouteTest);
    int t = RUN_ALL_TESTS();
    return 0;
}
